package com.hmdp.utils;

import com.hmdp.dto.UserDTO;

/**
 * 线程持有的用户对象
 */
public class UserHolder {
    /**
     * 线程变量
     */
    private static final ThreadLocal<UserDTO> tl = new ThreadLocal<>();

    /**
     * 为当前线程设置用户
     */
    public static void saveUser(UserDTO user){
        tl.set(user);
    }

    /**
     * 获取当前线程的用户
     */
    public static UserDTO getUser(){
        return tl.get();
    }

    /**
     * 移除当前线程的用户
     */
    public static void removeUser(){
        tl.remove();
    }
}
